import os

from langchain_core.messages import *
from langchain_openai import ChatOpenAI
OPENAI_API_KEY = os.environ["OPENAI_API_KEY_ZHIHU"]
OPENAI_API_BASE_ZHIHU = os.environ["OPENAI_API_BASE_ZHIHU"]

llm = ChatOpenAI(temperature=0.5, openai_api_key=OPENAI_API_KEY,openai_api_base=OPENAI_API_BASE_ZHIHU,model="gpt-4o")
messages=[
    SystemMessage(content="你是一个AI领域的专家，你以专家的身份回答用户的问题。"),
    HumanMessage(content="你好，我叫赵刚，我是一个AI的初学者"),
    AIMessage(content="您好，有什么我可以帮助您的。"),
    HumanMessage(content="您好，我叫什么名字？如何学习AI，请给我一个详细的指导。")
]
response = llm.invoke(messages)
print(response.content)